import os
import random


def get_random_jpg_files(target_dir, count=3):
    """
    递归获取指定目录及其子目录下的 count 个 JPG 文件。

    :param target_dir: 目录路径。
    :param count: 要获取的文件数量，默认为 3。
    :return: 一个包含文件路径的列表。
    """
    jpg_files = []

    # 递归遍历目录
    for root, _, files in os.walk(target_dir):
        for file in files:
            if file.lower().endswith(".jpg"):
                jpg_files.append(os.path.join(root, file))

    # 如果找到的jpg文件数量少于请求数量，调整count
    count = min(count, len(jpg_files))

    # 随机选择指定数量的jpg文件
    return random.sample(jpg_files, count) if jpg_files else []


if __name__ == "__main__":
    # 使用示例
    dir_path = r"/Users/wangchuanyi/mp4/pic"  # 替换为你的目录路径
    random_files = get_random_jpg_files(dir_path)
    for file_path in random_files:
        print(file_path)
